statistical operations on numpy array in python